from django.contrib.auth import authenticate, login, logout
from django.views import View
from django_redis import get_redis_connection

from apps.users.models import User
from django.http import JsonResponse, HttpResponse, HttpResponseRedirect
import json
import re


# Create your views here.
# 顺序：子模块.view > 子模块.urls > 父模块.urls
class UsernameCountView(View):
    """判断用户名是否重复注册"""

    def get(self, request, username):
        """
        :param request: 请求对象
        :param username: 用户名
        :return: JSON
        """
        count = User.objects.filter(username=username).count()
        return JsonResponse({'code': 0, 'count': count, 'errmsg': 'OK'})


class MobileCountView(View):
    """ 判断手机号是否重复注册 """

    def get(self, request, mobile):
        """
        :param request: 请求对象
        :param mobile: 手机号
        :return: JSON
        """
        count = User.objects.filter(mobile=mobile).count()
        return JsonResponse({'code': 0, 'errmsg': 'OK', 'count': count})


class RegisterView(View):
    """用户注册"""

    def post(self, request):
        """
        实现用户注册
        :param request: 请求对象
        :return: 注册结果
        """
        # 1.接收参数：请求体中的JSON数据 request.body
        json_bytes = request.body  # 从请求体中获取原始的JSON数据，bytes类型的
        json_str = json_bytes.decode()  # 将bytes类型的JSON数据，转成JSON字符串
        json_dict = json.loads(json_str)  # 将JSON字符串，转成python的标准字典
        # json_dict = json.loads(request.body.decode())

        # 2.提取参数
        username = json_dict.get('username')
        password = json_dict.get('password')
        password2 = json_dict.get('password2')
        mobile = json_dict.get('mobile')
        allow = json_dict.get('allow')
        sms_code_client = json_dict.get('sms_code')

        # 3.校验参数
        # 判断参数是否齐全
        if not all([username, password, password2, mobile, allow]):
            return JsonResponse({'code': 400, 'errmsg': '缺少必传参数!'})
        # 判断用户名是否是5-20个字符
        if not re.match(r'^[a-zA-Z0-9_]{5,20}$', username):
            return JsonResponse({'code': 400, 'errmsg': 'username格式有误!'})
        # 判断密码是否是8-20个数字
        if not re.match(r'^[0-9A-Za-z]{8,20}$', password):
            return JsonResponse({'code': 400, 'errmsg': 'password格式有误!'})
        # 判断两次密码是否一致
        if password != password2:
            return JsonResponse({'code': 400, 'errmsg': '两次输入不对!'})
        # 判断手机号是否合法
        if not re.match(r'^1[3-9]\d{9}$', mobile):
            return JsonResponse({'code': 400, 'errmsg': 'mobile格式有误!'})
        # 判断是否勾选用户协议
        if not allow:
            return JsonResponse({'code': 400, 'errmsg': 'allow格式有误!'})

        # 判断短信验证码是否正确：跟图形验证码的验证一样的逻辑
        # 提取服务端存储的短信验证码：以前怎么存储，现在就怎么提取
        redis_conn = get_redis_connection('code')
        sms_code_server = redis_conn.get('sms_%s' % mobile)  # sms_code_server是bytes
        # 判断短信验证码是否过期
        if not sms_code_server:
            return JsonResponse({'code': 400, 'errmsg': '短信验证码失效'})
        # 对比用户输入的和服务端存储的短信验证码是否一致
        if sms_code_client != sms_code_server.decode():
            return JsonResponse({'code': 400, 'errmsg': '短信验证码有误'})

        # 4.数据入库
        try:
            # 方法1：用户密码不会加密
            # user = User(username=username, password=password, mobile=mobile)
            # user.save()

            # 方法2：用户密码会加密
            user = User.objects.create_user(username=username, password=password, mobile=mobile)
        except Exception as e:
            return JsonResponse({'code': 400, 'errmsg': '注册失败!'})

        # 5.实现状态保持
        # 系统（Django）为我们提供了 状态保持的方法
        from django.contrib.auth import login

        # request, user,
        # 状态保持 -- 登录用户的状态保持
        # user 已经登录的用户信息
        login(request, user)

        # 6. 响应结果
        return JsonResponse({'code': 0, 'errmsg': 'ok'})


# 用户登录
class LoginView(View):
    def post(self, request):
        """
        用户登录
        :param request: 请求对象
        :return: 返回登录结果，JSON
        """
        # 1. 接收请求参数
        dict = json.loads(request.body.decode())
        username = dict.get('username')
        password = dict.get('password')
        remembered = dict.get('remembered')

        # 2. 校验（整体+单个）
        if not all([username, password]):
            return JsonResponse({'code': 400, 'errmsg': '请求参数不全'})
        # 3. 验证是否能够登录

        # 3.1 多账号登录：可以 用户名 或 手机号 登录
        import re
        if re.match('^1[3-9]\d{9}$', username):
            # 手机号
            User.USERNAME_FIELD = 'mobile'
        else:
            # account 是用户名
            # 根据用户名从数据库获取 user 对象返回.
            User.USERNAME_FIELD = 'username'

        # authenticate 为 Django 用户认证模块，认证ok时返回user对象，反之返回None
        user = authenticate(username=username, password=password)
        if user is None:
            return JsonResponse({'code': 400, 'errmsg': '用户名或密码错误'})
        # 4. 状态保持
        # login(request, user)
        # 5. 判断是否记住用户
        if remembered:  # 当选择记住用户，设置session过期时间为2周
            request.session.set_expiry(None)
        else:  # 当选择不记住用户，设置session为关闭浏览器时过期
            request.session.set_expiry(0)
        # 6. 返回JSON
        # 6.1 用户名写入到cookie
        response = JsonResponse({'code': 0, 'errmsg': 'ok'})
        # 注册时用户名写入到cookie，有效期15天
        response.set_cookie('username', user.username, max_age=3600 * 24 * 15)
        return response


# 退出登录
class LogoutView(View):
    def delete(self, request):
        """
        实现退出登录逻辑
        :param request: 请求对象
        :return: 返回结果 JSON
        """
        # 1. 清理session
        logout(request)
        # 2. 退出登录，重定向到登录页
        response = JsonResponse({'code': 0, 'errmsg': 'ok'})
        # 3. 退出时，清除cookie中的username
        response.delete_cookie('username')
        # 4. 返回结果
        return response


# 用户中心，也必须是登录用户
from utils.views import LoginRequiredJSONMixin


class CenterView(LoginRequiredJSONMixin, View):

    def get(self, request):
        """
        LoginRequiredMixin 未登录的用户 会返回 重定向。重定向并不是JSON数据
        我们需要是  返回JSON数据
        :param request: request.user 就是 已经登录的用户信息；是来源于 中间件
                        系统会进行判断 如果我们确实是登录用户，则可以获取到 登录用户对应的 模型实例数据
                        如果我们确实不是登录用户，则request.user = AnonymousUser()  匿名用户
        :return:
        """
        info_data = {
            'username': request.user.username,
            'email': request.user.email,
            'mobile': request.user.mobile,
            'email_active': request.user.email_active,
        }

        return JsonResponse({'code': 0, 'errmsg': 'ok', 'info_data': info_data})
